CV

sealed class CV : OnnxModelType<ImageRecognitionModel>

Image classification models.

Types

EfficientNet4Lite

android

object EfficientNet4Lite : ONNXModels.CV

Image classification model based on EfficientNet-Lite architecture. Trained on ImageNet 1k dataset. (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.labels method).

MobilenetV1

android

object MobilenetV1 : ONNXModels.CV

Image classification model based on MobileNetV1 architecture. Trained on ImageNet 1k dataset. (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.labels method).

Functions

model

android

open fun model(modelHub: ModelHub): OnnxInferenceModel

pretrainedModel

android

open override fun pretrainedModel(modelHub: ModelHub): ImageRecognitionModel

Properties

inputShape

android

open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath

android

open override val modelRelativePath: String

preprocessor

android

open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Inheritors

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV

ONNXModels.CV